Efficient method for KNN